Journals
  Publication Years
  Keywords
Search within results Open Search
Please wait a minute...
For Selected: Toggle Thumbnails
Pruning of YOLOv4 based on rank information in industrial scenes
Xiao QIN, Miao CHENG, Shaobing ZHANG, Lian HE, Xiangwen SHI, Pinxue WANG, Shang ZENG
Journal of Computer Applications    2022, 42 (5): 1417-1423.   DOI: 10.11772/j.issn.1001-9081.2021030448
Abstract231)   HTML14)    PDF (2320KB)(92)       Save

In the Radio Frequency IDentification (RFID) real-time defect detection task in industrial scenes, the deep learning target detection algorithms such as You Only Look Once (YOLO) are often adopted in order to ensure the detection precision and speed. However, these algorithms are still difficult to meet the speed requirement of industrial detection, and the corresponding network models cannot be deployed on resource-constrained devices. To solve these problems, the YOLO model must be pruned and compressed. A new network pruning method of the weighted fusion of feature information richness and feature information diversity based on rank information was proposed. Firstly, the unpruned model was loaded and reasoned, and the rank information of the corresponding feature maps of the filters was obtained in forward propagation to measure the feature information richness. Secondly, according to the different pruning rates, the rank information was clustered or the similarity of the rank information was calculated to measure the feature information diversity. Finally, the importance degrees of the corresponding filters were obtained after the weighted fusion and were sorted, and the filters with low importance were cut off. Experimental results show that, for YOLOv4, when the pruning rate is 28.87% and the weight of feature information richness is 0.75, the proposed method has the mean Average Precision (mAP) improved by 2.6% - 8.9% compared with the method that uses rank information of the feature maps alone, and the model pruned by the proposed method even has the mAP increased by 0.4% and the model parameters reduced by 35.0% compared with the unpruned model, indicating that the proposed method is conducive to the model deployment.

Table and Figures | Reference | Related Articles | Metrics